package com.dji.sample.common.util;


import java.io.*;
import java.nio.file.FileVisitResult;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.SimpleFileVisitor;
import java.nio.file.attribute.BasicFileAttributes;
import java.util.Objects;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

/**
 * @author yjt
 * @date 2022-10-09
 * @description:
 */
public class FileUtil {

    public static void zipFile(File fileToZip, String fileName, ZipOutputStream zipOut) throws IOException {
        if (fileToZip.isHidden()) {
            return;
        }
        if (fileToZip.isDirectory()) {
            if (fileName.endsWith("/")) {
                zipOut.putNextEntry(new ZipEntry(fileName));
                zipOut.closeEntry();
            } else {
                zipOut.putNextEntry(new ZipEntry(fileName + "/"));
                zipOut.closeEntry();
            }
            File[] children = fileToZip.listFiles();
            for (File childFile : children) {
                zipFile(childFile, fileName + "/" + childFile.getName(), zipOut);
            }
            return;
        }
        FileInputStream fis = new FileInputStream(fileToZip);
        ZipEntry zipEntry = new ZipEntry(fileName);
        zipOut.putNextEntry(zipEntry);
        byte[] bytes = new byte[1024];
        int length;
        while ((length = fis.read(bytes)) >= 0) {
            zipOut.write(bytes, 0, length);
        }
        zipOut.closeEntry();
        fis.close();
    }



    /**
     * 将存放在sourceFilePath目录下的源文件,打包成fileName名称的ZIP文件,并存放到zipFilePath。
     * @param sourceFilePath 待压缩的文件路径
     * @param zipFilePath    压缩后存放路径
     * @param fileName       压缩后文件的名称
     * @return flag
     */
    public static boolean fileToZip(String sourceFilePath, String zipFilePath,
                                    String fileName) {
        boolean flag = false;
        File sourceFile = new File(sourceFilePath);
        FileInputStream fis = null;
        BufferedInputStream bis = null;
        FileOutputStream fos = null;
        ZipOutputStream zos = null;
        if (!sourceFile.exists()) {
            System.out.println(">>>>>> 待压缩的文件目录：" + sourceFilePath + " 不存在. <<<<<<");
        } else {
            try {
                File zipFile = new File(zipFilePath + "/" + fileName + ".zip");
                if (zipFile.exists()) {
                    System.out.println(">>>>>> " + zipFilePath + " 目录下存在名字为："
                            + fileName + ".zip" + " 打包文件. <<<<<<");
                } else {
                    File[] sourceFiles = sourceFile.listFiles();
                    if (null == sourceFiles || sourceFiles.length < 1) {
                        System.out.println(">>>>>> 待压缩的文件目录：" + sourceFilePath
                                + " 里面不存在文件,无需压缩. <<<<<<");
                    } else {
                        fos = new FileOutputStream(zipFile);
                        zos = new ZipOutputStream(new BufferedOutputStream(fos));
                        byte[] bufs = new byte[1024 * 10];
                        for (File file : sourceFiles) {
                            if(file.isDirectory()){
                                zos.putNextEntry(new ZipEntry(file.getName()+"/"));
                                for (File listFile : Objects.requireNonNull(file.listFiles())) {
                                    // 创建ZIP实体,并添加进压缩包
                                    ZipEntry zipEntry = new ZipEntry(file.getName()+"/"+listFile.getName());
                                    zos.putNextEntry(zipEntry);
                                    // 读取待压缩的文件并写进压缩包里
                                    fis = new FileInputStream(listFile);
                                    bis = new BufferedInputStream(fis, 1024 * 10);
                                    int read = 0;
                                    while ((read = bis.read(bufs, 0, 1024 * 10)) != -1) {
                                        zos.write(bufs, 0, read);
                                    }
                                    try {
                                        fis.close();
                                        bis.close();
                                    }catch (IOException e){}
                                }
                                continue;
                            }
                            // 创建ZIP实体,并添加进压缩包
                            ZipEntry zipEntry = new ZipEntry(file.getName());
                            zos.putNextEntry(zipEntry);
                            // 读取待压缩的文件并写进压缩包里
                            fis = new FileInputStream(file);
                            bis = new BufferedInputStream(fis, 1024 * 10);
                            int read = 0;
                            while ((read = bis.read(bufs, 0, 1024 * 10)) != -1) {
                                zos.write(bufs, 0, read);
                            }
                            try {
                                fis.close();
                                bis.close();
                            }catch (IOException e){}

                        }
                        zos.finish();
                        zos.closeEntry();
                        zos.close();
                        flag = true;
                    }
                }
            } catch (IOException e) {
                e.printStackTrace();
            } finally {
                // 关闭流
                try {
                    if (null != bis) {
                        bis.close();
                    }
                    if (null != zos) {
                        zos.close();
                    }
                    if(null!=fis){
                        fis.close();
                    }
                    if(null!=fos){
                        fos.close();
                    }
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
        return flag;
    }


    public static void deleteFile(String path) {
        for (File listFile : Objects.requireNonNull(new File(path).listFiles())) {
            try {
                if (listFile.isDirectory()) {
                    if (listFile.length() > 0) {
                        for (File file : Objects.requireNonNull(listFile.listFiles())) {
                            Files.delete(file.toPath());
                        }
                    }
                    continue;
                }
                Files.delete(listFile.toPath());
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
        new File(path).delete();
    }


}
